package com.example.mq.mqserver.core;

import com.example.mq.common.ConsumerEnv;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.Data;

import java.util.*;
import java.util.concurrent.atomic.AtomicInteger;

// MSG => Message
// 该类表示一个存储消息的队列
@Data
public class MSGQueue {
    // name, durable, exclusive, autoDelete, arguments

    // 表示队列的身份标识
    private String name;
    // 表示队列是否持久化
    private boolean durable = false;

    // TODO 独占功能, true-该队列只能被一个消费者使用, false-非独占,该队列可被所有消费者使用
    private boolean exclusive = false;
    // TODO 自动删除功能, true-无人使用会自动删除, false-无人使用不会自动删除
    private boolean autoDelete = false;
    // TODO 拓展参数
    private Map<String, Object> arguments = new HashMap<>();

    // 记录当前队列有哪些消费者订阅了.
    private List<ConsumerEnv> consumerEnvList = new ArrayList<>();
    // 记录当前取到了第几个消费者. 方便实现轮询策略
    private AtomicInteger consumerSeq = new AtomicInteger(0);

    // 添加一个新的订阅者
    public void addConsumerEnv(ConsumerEnv consumerEnv) {
        consumerEnvList.add(consumerEnv);
    }

    // TODO 删除一个订阅者

    // 挑选一个订阅者来处理当前的消息. (轮询的方式)
    public ConsumerEnv chooseConsumer() {
        if(consumerEnvList.size() == 0) {
            // 该队列没有人订阅
            return null;
        }
        // 计算一下当前要取的元素的下标
        int index = consumerSeq.get() % consumerEnvList.size();
        consumerSeq.getAndIncrement();
        return consumerEnvList.get(index);

    }

    public String getArguments() {
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            return objectMapper.writeValueAsString(this.arguments);
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
        return "{}";
    }

    public void setArguments(String argumentsJson) {
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            // 匿名内部类写法
            this.arguments = objectMapper.readValue(argumentsJson, new TypeReference<Map<String, Object>>() {});
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
    }

//---------------------------------------------------------

    public Object getArguments(String key) {
        return arguments.get(key);
    }

    public void setArguments(String key, Object value) {
        arguments.put(key, value);
    }

    public void setArguments(Map<String, Object> arguments) {
        this.arguments = arguments;
    }

}
